package com.ctrip.corp.obt.shard.sql.mybatis.typehandler;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import com.ctrip.corp.obt.generic.utils.NumberUtils;
import org.apache.ibatis.type.MappedJdbcTypes;
import org.apache.ibatis.type.MappedTypes;

/**
 * @author x.shen
 * @desc 类型转换器，实现jdbcType=tiny转Java=Boolean
 * @since 2023/10/30
 */
@MappedTypes(Boolean.class)
@MappedJdbcTypes(JdbcType.TINYINT)
public class TinyIntToBooleanTypeHandler extends BaseTypeHandler<Boolean> {
    /**
     * 插入时设置参数类型，用于将JAVA类型参数设置到预处理语句PreparedStatement中
     * @param ps
     * @param i
     * @param parameter
     * @param jdbcType
     * @throws SQLException
     */
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Boolean parameter, JdbcType jdbcType) throws SQLException {

    }

    /**
     * 获取时转换回的自定义类型，根据列名获取，用于将查询结果集ResultSet的columnName列的数据转换成JAVA类型（JdbcType -> JavaType）
     * @param rs
     * @param columnName
     * @return
     * @throws SQLException
     */
    @Override
    public Boolean getNullableResult(ResultSet rs, String columnName) throws SQLException {
        String value = rs.getString(columnName);
        return convertToBoolean(value);
    }

    /**
     * 根据索引位置获取，用于将查询结果集ResultSet的第columnIndex列的数据转换成JAVA类型（JdbcType -> JavaType）
     * @param rs
     * @param columnIndex
     * @return
     * @throws SQLException
     */
    @Override
    public Boolean getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        String value = rs.getString(columnIndex);
        return convertToBoolean(value);
    }

    /**
     * 根据存储过程获取
     * @param cs
     * @param columnIndex
     * @return
     * @throws SQLException
     */
    @Override
    public Boolean getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        String value = cs.getString(columnIndex);
        return convertToBoolean(value);
    }

    private Boolean convertToBoolean(String value) {
        if (NumberUtils.isNumber(value)) {
            return Integer.parseInt(value) == 1;
        }
        return null;
    }
}
